How to create 3D array in Python using numpy